Updated widget logic