ADding remove functionality