go fmt code a bit