Added pset3 code