Add a simpler implementation for removing element problem.