break out unit test class. Add another test case