Finished adding the remove method