edit main.go