add models
bug fix