Prevent deprecation warnings by switching .type -> .class
Merge pull request #7 from tas50/master

Prevent deprecation warnings by switching .type -> .class