Implemented JUnit test for the new user story