Fix coloured output, simplify exits on error