Create pset1part2.py