made some fixes to validation part of pruning, not exactly getting the best scores