Added a test, go fmt, and tiny fixes.