Add direct mongo update support