fixed #11 - no need to wrap the class, only to add the model
Merge pull request #12 from avishai-ish-shalom/issue-11

fixed  #11 - no need to wrap the class, only to add the model