small changes in the budget, refactor