Cleaned up the budget controller.