Added dataset.take_columns()