Nit, but please write as `idx += len(self.items_w)`