Factored out cross validation to it's own notebook