Add cross validation
Working