Add optimal_column_to_split for simplicity