this fixes #9 and #10
Merge pull request #24 from RedTurtle/fix_label_class_problem

this fixes #9 and #10