add docstrings for classes
added TODO list