Add sort_by_key