Update main.go

add format flag