Correct toggle behavior and styling