Fixes #2
Reindent code