add tests for remaining models