Two Dimensional Array Code In Python