Project 2 completed in Python