Update remove-element.cpp