Added task model assignment