Completed remove function. All 4 tests passing.