SA budget to upgrade railway stations