Refactor: do not need to keep the document
add more to sample
Add more test cases