Found that WhichIsMax can use which to return indices