Add success message
Fix for NPE
Change message
Update message