Adding in color output