diff --git a/states.h b/states.h
--- a/states.h
+++ b/states.h
@@ -27,6 +27,8 @@ typedef union
         uint32_t ignore_error;
         int32_t setpoint_brew;
         int32_t setpoint_steam;
+        uint32_t control_mode;
+        uint32_t plant_type;
     } val;
 
     uint16_t data[128];
@@ -39,7 +41,9 @@ enum tempunits {
 
 enum state {
     STATE_IDLE = 0,
-
+    
+    STATE_SETMODE,
+    STATE_SETPLANTTYPE,
     STATE_SETP,
     STATE_SETI,
     STATE_SETD,
@@ -56,6 +60,16 @@ enum state {
 	STATE_RESET,
 };
 
+enum control_mode {
+    MODE_PID = 0,
+    MODE_THERMOSTAT,
+};
+
+enum plant_type {
+    PLANT_HEATER = 0,
+    PLANT_COOLER,
+};
+
 enum GOTO_MODE {
 	#ifdef BOOTLOADER_SHORTCUT
 	MODE_BOOTLOADER,