Adding support to use YARN for provisioning