Added comment to problem 4