Create ProblemSet2.py