Add argument to sort_table