Updating comments in problem1.py