Fix code example
Merge pull request #25 from aebm/patch-3

Fix code example