Improving train plan