Prime number generator algorithm python