clean

clean the train classes