Adding remove functionality to tasks