Added color variables and cleaned up code.