pset4, problem 1, done