Problem 003 add comments