Added in testing for removing a task