Index: firmware/App/Services/SystemComm.c
===================================================================
diff -u -rd89f4eb7da7574306f45eefea5a2fc3b980fd7c9 -rc41e74ff80145e90c7066a5970c149c2ed4d59dd
--- firmware/App/Services/SystemComm.c	(.../SystemComm.c)	(revision d89f4eb7da7574306f45eefea5a2fc3b980fd7c9)
+++ firmware/App/Services/SystemComm.c	(.../SystemComm.c)	(revision c41e74ff80145e90c7066a5970c149c2ed4d59dd)
@@ -191,7 +191,12 @@
 void checkInFromDG( void )
 {
     dgIsCommunicating = TRUE;
-    timeOfLastDGCheckIn = getMSTimerCount();
+    timeOfLastDGCheckIn = getMSTimerCount();
+
+    if ( TRUE == isAlarmActive( ALARM_ID_DG_COMM_TIMEOUT ) )
+    {
+        clearAlarm( ALARM_ID_DG_COMM_TIMEOUT );
+    }
 }
 
 /*********************************************************************//**
@@ -1184,8 +1189,8 @@
             handleTreatmentParametersFromUI( message );
             break;
 
-        case MSG_ID_STARTING_STOPPING_TREATMENT_CMD:
-            handleUIStartTreatmentMsg( message );
+        case MSG_ID_UI_INITIATE_TREATMENT_REQUEST:
+            handleInitiateTreatmentRequest( message );
             break;
 
         case MSG_ID_UI_USER_CONFIRM_TREATMENT_PARAMS:
@@ -1216,6 +1221,38 @@
             handleTreatmentEndCmd( message );
             break;
 
+        case MSG_ID_UI_SAMPLE_WATER_CMD:
+            handleSampleWaterCmd( message );
+            break;
+
+        case MSG_ID_UI_SAMPLE_WATER_RESULT:
+            handleSampleWaterResult( message );
+            break;
+
+        case MSG_ID_UI_CONSUMABLE_INSTALL_CONFIRM:
+            handleConsumableInstallConfirm( message );
+            break;
+
+        case MSG_ID_UI_INSTALLATION_CONFIRM:
+            handleInstallationConfirm( message );
+            break;
+
+        case MSG_ID_UI_START_PRIME_REQUEST:
+            handleStartPrimeCmd( message );
+            break;
+
+        case MSG_ID_UI_PATIENT_CONNECTION_BEGIN_REQUEST:
+            handleContinueToTreatmentCmd( message );
+            break;
+
+        case MSG_ID_UI_PATIENT_CONNECTION_CONFIRM:
+            handlePatientConnectionConfirmCmd( message );
+            break;
+
+        case MSG_ID_UI_START_TREATMENT_REQUEST:
+            handleStartTreatmentRequest( message );
+            break;
+
         case MSG_ID_DG_COMMAND_RESPONSE:
             handleDGCmdResp( message );
             break;