codility solutions python