add better go code