Adapted the code for colors

Signed-off-by: Frederick Eyland <freeyland@gmail.com>