Add color output for messages