Add support for task-specific workers