adding some documentation

slight refactor of unit tests to prepare for more