Update push registration code