Fixing the code so that it works
Merge pull request #131 from AlexCogn/patch-1

Fixing the code so that it works