Setup testing. Add some tests for setModels