Added a JUnit test