fix the exercise 5.4
Merge pull request #8 from evan617/master

fix the exercise 5.4