Implemented JUnit test