Added support for findOneAndUpdate, findByIdAndUpdate