I’m trying a slight variant of this on LM, let me know how it works.