Add tests on models classes