Added a done callback