Add comments to Python version
Add comments to Go version