Update main.go

Fixes formatting