Adding train classes