removed reference to KFold