Adding remove functionality