Updated to new model variable
Merge pull request #22 from adaptlearning/issue/#19

Updated to new model variable