Update addon.lua

Update to the new functions.