+ Array Generation: Refactored to use list comprehension
+ Refactored to be PEP8 compliant