Allow diferent colorschemes
Add tests