Added basic model setup