Create pset1part1.py