Comments added to problem 4

Comments added to problem 4