Add pygment support