Update main.go

compiler error fixed