Wrote more idiomatic Go after reading Jeff's comments
go fmt