Add done callback