Update main.go

Formatted code