Update main.go