Fix formatting
Merge pull request #2 from rbone/bugs/formatting_fix

Fix formatting