Fix a comment and a apply few gofmt changes.