Sorry I had to go offline @boazmohar I can help you think through this over the weekend (or better next Monday) I'm pretty sure you just want to add another constructor arg to your dataset and then access that via self.