fix class names
Merge pull request #5 from tmf/master

fix class names