working on test class