Changes the behaviour of model's remove function to give feedback in ui