add models
fix bug