Updated user interface to include budget calculations