Update code to use get_color_theme()