Add the test class

Added test class