Added the remove method