Cleared code, PEP-8 compatible, list comprehension