Add colored output, print output at end of run