Add support for setup task