Update how extension works