Cleaned up the display