Modified the budget view with edit button