added 'update budget' feature and made it pass