initial code for pset 1