Update ProblemSet1.1.py

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