Create problem_set_2.py