Add register push notification function