Add some docstrings and a bit more README
More documentation cleanup.