I am probably overthinking a bit here. But see below for a solution. I can publish this to Rhythm as well.