corrects names of classes
Merge pull request #5 from bhattigurjot/master

corrects names of classes