Add support for main display