Change of train() definition