Modified p01_test_1 to match Coursera's suggested test