Added JUnit test class