WIP problem 4
Run everything through `go fmt`