I am still working on the legend and having it do exactly like you said. I have to change things up a bit but hopefully I can be able to implement it soon