Add sort feature