From cfe4dd1c59f1d2c62bfcc7380cc08d151368301d Mon Sep 17 00:00:00 2001
From: estevez-dev <vyalov.egor@gmail.com>
Date: Wed, 16 Oct 2019 19:39:14 +0300
Subject: [PATCH] Vacuum state colors update

---
 lib/const.dart                        | 1 +
 lib/entities/entity_colors.class.dart | 2 ++
 2 files changed, 3 insertions(+)

diff --git a/lib/const.dart b/lib/const.dart
index 002c052..8568548 100644
--- a/lib/const.dart
+++ b/lib/const.dart
@@ -33,6 +33,7 @@ class EntityState {
   static const docked = 'docked';
   static const returning = 'returning';
   static const error = 'error';
+
 }
 
 class EntityUIAction {
diff --git a/lib/entities/entity_colors.class.dart b/lib/entities/entity_colors.class.dart
index d226336..7ff35ec 100644
--- a/lib/entities/entity_colors.class.dart
+++ b/lib/entities/entity_colors.class.dart
@@ -18,6 +18,8 @@ class EntityColor {
     "above_horizon": Colors.amber,
     EntityState.home:  Colors.amber,
     EntityState.open:  Colors.amber,
+    EntityState.cleaning:  Colors.amber,
+    EntityState.returning:  Colors.amber,
     EntityState.off: defaultStateColor,
     EntityState.closed: defaultStateColor,
     "below_horizon": defaultStateColor,