Cross validation working. Need pruning.