Rewamp widget's logic