Fixed wrong class names
Merge pull request #9 from madskristensen/patch-1

Fixed wrong class names