Update split_to_prime.py