fix the exercise 5.4
Merge pull request #43 from pezy/upstream

Improved : exercise 5.4