UI fixes for display page