Added test for remove in the middle