README fixes

This fixes the code snippets in the README and cleans them up a little (`gofmt`).