fixed formatting in list comprehension