__getitem__ now working nicely