Updating pset1 sol