Update array_utils.py