Add widget functionality