added a few tests to model