Added in pipeline and cross validation