diff --git a/Week1/prep-exercises/1-traffic-light/traffic-light-1.js b/Week1/prep-exercises/1-traffic-light/traffic-light-1.js
index f1d9169..982af8a 100644
--- a/Week1/prep-exercises/1-traffic-light/traffic-light-1.js
+++ b/Week1/prep-exercises/1-traffic-light/traffic-light-1.js
@@ -11,21 +11,12 @@ let rotations = 0;
 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
 }
 
-/**
- * The output should be:
+if (currentState === "green") { trafficLight.state = "orange"; 
 
-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
+} else if (currentState === "orange") { trafficLight.state = "red";
 
-*/
+ } else if (currentState === "red") { trafficLight.state = "green";
+   rotations++; 
+  }
\ No newline at end of file
diff --git a/Week1/prep-exercises/1-traffic-light/traffic-light-2.js b/Week1/prep-exercises/1-traffic-light/traffic-light-2.js
index 8c6ba95..5429fbf 100644
--- a/Week1/prep-exercises/1-traffic-light/traffic-light-2.js
+++ b/Week1/prep-exercises/1-traffic-light/traffic-light-2.js
@@ -1,9 +1,5 @@
 "use strict";
-/**
- * The `possibleStates` property define the states (in this case: colours)
- * in which the traffic light can be.
- * The `stateIndex` property indicates which of the possible states is current.
- */
+
 const trafficLight = {
   possibleStates: ["green", "orange", "red"],
   stateIndex: 0,
@@ -14,20 +10,14 @@ while (cycle < 2) {
   const currentState = trafficLight.possibleStates[trafficLight.stateIndex];
   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 cycles and turn it green
-}
-
-/**
- * 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
+  if (currentState === "green")
+     {  trafficLight.stateIndex = 1; 
+      } else if (currentState === "orange") 
+        {  trafficLight.stateIndex = 2 ;
 
-*/
+        } else if (currentState === "red") { 
+         trafficLight.stateIndex = 0; 
+         
+         cycle++;
+         } 
+        }
\ No newline at end of file