Updated init.lua to make callback registration expandable
