updated comments in problem4