Add support for display setup script