Update Output 2D array.py