Improved notification styling and code structure