Add callback fn
Add docs