UI Improvements for Display Toggle