fixed the remove feature