Update widget to new design