This is the problem1.py that I forgot to add and commit when I submitted
problems 2, 3, and 4"