[TASK] Flatten Model