Fix code style issues
Merge pull request #4 from javigomez/speed-fix

Fix code style issues