[FIX] Change the way to store documents in DB

- Used db.collection.find_and_modify instead of document.save