Adding test case for remove method
Adding remove method