I hate ```np.where``` - I can't ever remember not wanting the first index from the result...
Anyway - this where is fine.