Add focused state for ImageButton component