Completed problem1.py