Changes to try and stop problem where display disappears.