Python:Add for else & list comprehension