Proposal

Because you liked my first suggestion, I took the liberty of suggesting some more changes to simplify readability. Hope they are welcome and I'd love to hear what you think of them. I also hope I didn't break anything :100: 
Thanks for clarifying, shouldn't this be a configuration function passed to Start() ?