Change highlighting to pygments