Tried to finish the remove method, but there are problems.