model now simulating well and test pass