Added helper for display logic