Implementing 2d Array In Python