Would be great to be able to label as well search tasks

Additionally it would be nice to know the model that was used for the task and warn if you are trying to use a different model than when you started with.