Moved colorization into its own function and used it across the board.