Adding a finished callback