Adding tests for document and fixing things