add models
fix for production