Refactored widget display listener with new widget system