add notification style and arrange code