I think I finished the python part of the assignment. Just need to comment