adjust the class names to match what is there
Merge pull request #7 from jeremyBass/master

adjust the class names to match what is there