Format code for consistency and improve readability in main.go

Format code for consistency and improve readability in main.go #206