fixed the homework based on feedbacks that I got
Merge pull request #79 from Woonyung/master

second pull request