feature: updated models and migrations