Updated widget JS