add models
fix