pset6 code