Add query_fields to model schema