Add sorting feature