Enable indexing into data object like a numpy array

Using the getitem method, we should be able to do the following:
```
print(data[i])
```
Rather than
```
print(data.to_numpy()[i])
```
Which makes a lot of the code neater. Potentially also enable the setitem feature, but that might be dubious