Revert "Revert "Added Completion Block to Close Button""

This reverts commit b3a794e6919beb56393e4e7ca8b0f3f9c10dd46d.