enhanced the test class