problem_1.py is ostensibly working at this point