Add sieve of atkin in python