Change model use to a proper way in terms of MVC