Added an option to color out the ouput