Filtering prime numbers using list comprehension