Added display on/off control