Update main.go

1