Added more tests for the document