Add a change as requested in homework.
Merge pull request #10 from kourosh/master

Add a change as requested in homework.