fix collapse button
Finish menu_button_on_small_devices