Fixed identation and included new test under existing context