Updating custom functionality