Add support for $setOnInsert