Option to have a custom sort function

[stev-0]

This might help with #398

There are probably better ways to do this, but this worked for my simple case