Added pygments for syntax highlighting