Fix class name
Merge pull request #13 from lchenay/version2

Fix class name