Make removeOne method consume ScriptableDocument.

Currently, removeOne method only consumes a query. This patch makes the method
also consume a document.