Removed redundant code
Added dependency for junit in pom