list comprehension for prime numbers