Change model extenstions to support class_name for tenant