Still fixing if statement in display.