gofmt

And added another test