attempt at pset61 fix