Added models and changed the environment to use different code base