Update main.go

Fixed a few more errors.