Added GetBudget extension method