@@ -1,9 +1,4 @@
-"use strict";
-/**
- * The `state` property says what the traffic light's state (i.e. colour) is at
- * that moment.
- */
-const trafficLight = {
+let trafficLight = {
   state: "green",
 };
 
@@ -12,20 +7,14 @@ while (rotations < 2) {
   const currentState = trafficLight.state;
   console.log("The traffic light is on", currentState);
 
-  // TODO
-  // if the color is green, turn it orange
-  // if the color is orange, turn it red
-  // if the color is red, add 1 to rotations and turn it green
+  if (currentState === "green") {
+    trafficLight.state = "orange";  
+  }
+  else if (currentState === "orange") {
+    trafficLight.state = "red";
+  }
+  else if (currentState === "red") {
+    trafficLight.state = "green";
+    rotations++;
+  }
 }
-
-/**
- * The output should be:
-
-The traffic light is on green
-The traffic light is on orange
-The traffic light is on red
-The traffic light is on green
-The traffic light is on orange
-The traffic light is on red
-
-*/