updated notification part
Merge pull request #5 from kelvinzhu2000/yanzhang

updated notification part