Toggle mode fix: Changed ImageButton to ToggleButton