Add support for storage setup callbacks