Add support for afterBoot callback