Proofread and cleaned the test class