modify main.go