Add ANSI coloring of diagram when outputting to terminal

Signed-off-by: Johan Herland <johan@herland.net>