modify task to use all models