add train class