Looks like you added, what I suggested before, and it looks cool too.

I have a suggestion about the toggle button, Here's how it looks now



Toggle button is not that much visible unless we give a closer look to it, so try adding a border with a bolder look or even a different color can work fine.