Added support for Set Budget