Implementation of remove method added
JUnit test updates