Add tests of models