Add callback to load function