Solved pset 3!!!