Changed the way to import models
Import models only to controllers