Added JUnit Class