Refactor budget