Improve gocode support