Added basic model tests