Python Program Using List Comprehension