edit to problem 1 hw3