modify ctrain.pyx