Modify cross-validation.py