fixed missing remove functionality