Fixed display issues

Display did not light up all the time.