fix typo in docstring
more work on styling