Fix for some whitespace problems.

- Fixes #3