add models (not perfect)