Added color variables
Fixing output