Added support for more than just red color