Add colorizing functionality