Minor idiomatic fixes

Nice package! Short and sweet. This PR just makes some very minor style and idiomatic changes. I hope it's helpful.