Added models to models_setup.py