same idea, yes

this seems simpler and more straight forward though