Improving code colorization