Idiomatic improvments

Just some tweaks to make it more like the rest of the Go packages. Open to suggestions.