Add open/close button. Thanks @dbhurely