fixed indentation and tests for classes