Add mapping for sorting