minor fix in Pset4