Fix some styles
After using gofmt -w