Edited budget function