add tests to new models