Changed the JUnit class