add train class and spec