modify for python and go