missing class, fixed!
Merge pull request #25 from AmirolAhmad/master

missing class, fixed!