Issue: How to use arrays in python