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

fix class names