Having problems wiht numpy & getting an error of too many indicies