class Traindiagram