switch to not indexing samples like a data frame

It looks like the switch away from using data frames broke this example. I think this works as intended now.