added document_test.go, made it work