Add pygments for better highlighting