pygments highlight function fixed, but not very well