Add support for display