Added JUnit test