Adding sort feature