Add optional sorting