Fix a pretty substantial typo...
Indices need to be numpy arrays
Whitespace / PEP8 cleanup