Adding some simple model tests