improve code highlighting styles