That actually works. Thanks a lot!

A bit of a weird API forcing it to be only the first column, but I can get around it. Also seems to be reset when the model is reset which is also annoying.