Add colored code highlighting