Fix code example
Merge pull request #6 from palmerabollo/master

Minor: fix code example to make it clearer