refactored and added a testModels file