Add afterTest closure to get notified after test execution