Minor updates to go.py