Attempt to add pygments for highlighting