Improved display a little