fix class naming issue
Merge pull request #6 from JonMcPartland/patch-1

fix class naming issue