Update model to use MongoDB's ID