added some work on the budget