--- a/iaupdate/IAD/ui/inc/iaupdateautomaticcheck.h	Tue Jun 15 13:36:10 2010 +0100
+++ b/iaupdate/IAD/ui/inc/iaupdateautomaticcheck.h	Thu Jul 22 16:35:01 2010 +0100
@@ -1,5 +1,5 @@
 /*
-* Copyright (c) 2008-2009 Nokia Corporation and/or its subsidiary(-ies).
+* Copyright (c) 2008-2010 Nokia Corporation and/or its subsidiary(-ies).
 * All rights reserved.
 * This component and the accompanying materials are made available
 * under the terms of "Eclipse Public License v1.0"
@@ -22,13 +22,17 @@
 
 // INCLUDES
 #include <e32base.h>
+#include "iaupdatedialogobserver.h"
 
+class HbAction;
+class IAUpdateDialogUtil;
 // CLASS DECLARATION
 /**
 *
 */
 
-class CIAUpdateAutomaticCheck : public CBase
+class CIAUpdateAutomaticCheck : public CBase,
+                                public IAUpdateDialogObserver
     {
 public:
 
@@ -59,7 +63,7 @@
     * @return True value if automatic update connections are allowed, EFalse if not allowed.
     *         Returns always true if this function is called after first time
     */
-    TBool AcceptAutomaticCheckL();
+    void AcceptAutomaticCheckL();
     
     TBool AutoUpdateCheckEnabledL();
         
@@ -79,10 +83,15 @@
     * Enables automatic updates cheks
     */
     void EnableAutoUpdateCheckL( TBool aEnable );
+
+private: // From IAUpdateDialogObserver     
+         
+     void dialogFinished(HbAction *action);       
     
         
 private: //data
-    
+    IAUpdateDialogUtil *mDialogUtil;
+    HbAction *mPrimaryAction;
     };