Fixed issue #2 and indentation