Added problem 2 in pset 4