Add highlighting for go