pset6 prob2 Update