trying to fix pset2