cleaned up the comments for problem3