Update formatting with go fmt