Refactor warning display logic to Warning module