Improve documentation and run gofmt