Update main.go

finally