pset6 prob1 Complete