Added 2nd train functionality