updated comment docs
unit tests to work more code
unit tests