Updated Remove Function