Ran go fmt. Just some minor cleanups.