Refactor. Swap buttons for a toggle