Add some additonal functionality to clay.stats

Shamelessly cribbed from @dnathe4th
PEP 8 fix
Add docstrings, fix import order
Refactor the stats module a bit more