Add a sort function