Added a test class

PR test