Add colorful output support