Simple 2d Array Program In Python