Adding correct widget