Add support for desired task count