Can use list comprehension here
python convention change
Constants should be in caps