Added test class