Add initializeCellWidget and uninitializeCellWidget