Added support for different color schemes