Python Function Finding Prime