Update HW 6

small edits to problem 3