Added JUnit class