Add pygments support