Minor adjustments

I've processed the code through `gofmt`, and made some tweaks to the comments. If you would rather keep the formatting as it is, feel free to close the issue. If you'd like to only use some of the changes, let me know and I'll revert the rest.

Thanks for the koans, they've been a lot of help in practicing the language :).