Updated solution to pset3