<SYSTEM_TASK:>
Solve the following problem using Python, implementing the functions described below, one line at a time
<END_TASK>
<USER_TASK:>
Description:
 def _toolkits_select_columns(dataset, columns):
 """ Same as select columns but redirect runtime error to ToolkitError. """

    try:
        return dataset.select_columns(columns)
    except RuntimeError:
        missing_features = list(set(columns).difference(set(dataset.column_names())))
        raise ToolkitError("Input data does not contain the following columns: "  +
                "{}".format(missing_features))