Prime nth number in Python Functions added