Added doneClick event listener