Fixed problem 4 comment issue