gofmt and clean up tests