Add push notification function