Corrected DisplayWidget class