Budget can end the wait for rail upgrades