pset4 fixes