Use augment instead of Class for inheritance