Updated unit test and pom