Adding a button to toggle display on/off