Small updates to colorscheme