Added the JUnit Tests