Cross validation added. (not complete yet!)