Adding predictions, changing split method.