fixed the remove method