That's a good idea but it works from the change in context no? So you lose context from your big model. It's a good suggestion though.