revise sorting python