Create problem_set4.py