Added tests for remove method.
Implemented remove method.