add model tests