fixed some docstrings and added class docstring
more documenting and added some idea for later work