Update ProblemSet1.2.py

Adding a comment that this is the code used to pass the test on the course website