Updated cross_validation.py