Adding a list comprehension test script