cleaned up comments n docstrings