minor refactoring
re-add test document