Update cloud code for notification feature