fixed a bug and added colorscheme