Update class references
Merge pull request #3 from mfairchild365/master

Update class references