Added models list and some edits to model code