Update ilatrain.py