Updated code for push notification