Added the edit budget functionality

Tried following Graham's style as much as I could!