Add model setup