Added pygments highlighting for code