Adding a test, and cleaning up indentation