Add ANSI code highlighting