Budget focuses on rail safety