Updated models

Updated main model