Add some model tests