Adding sorting algos in python