Fixed the count issue of the class
Merge pull request #255 from sheebans/master

Fixed the count issue of the class