Add ability to get class for augmented data
Fixed joining on object properties