Added the JUnit Test