While Loops and Prime Numbers using Python