Add if around colorscheme