Update main.go

fixed compile erros