update main.go