'Set color scheme for syntax highlighting'

def set_color_scheme(self, color_scheme):

    self.color_scheme = color_scheme
    if (self.highlighter is not None):
        self.highlighter.set_color_scheme(color_scheme)
        self._apply_highlighter_color_scheme()
    if self.highlight_current_cell_enabled:
        self.highlight_current_cell()
    else:
        self.unhighlight_current_cell()
    if self.highlight_current_line_enabled:
        self.highlight_current_line()
    else:
        self.unhighlight_current_line()
