formatting

run gofmt and improve the test (in-progress)