Add inital model tests