Fix mongo update
Update model to take in account membership types