Wait for end (@Rahul)