Refactored budget.rb.