tms: Add support for colorized output

Colorized output can be easier on the eye, so make it easy to add to
commands.

Signed-off-by: Thierry Reding <treding@nvidia.com>