Cleanup
Use Pygments API