It might be helpful to include your code in a class (or several classes) and give good names to functions. Make sure put comments in """ """ for Python instead of after #. Also, write good test cases. Seems the interviewer really cares about the details.