insert the model class
Merge pull request #2 from dilini-mampitiya/master

insert the model class