add tests, gofmt stuff