Improve the docs, adding a better overall explanation
Add optimal_column_to_split for simplicity