Problem Getting Numpy Matrix