diff --git a/docs/diagrams/src/motor_state.wsd b/docs/diagrams/src/motor_state.wsd
index b56e243..862e45c 100644
--- a/docs/diagrams/src/motor_state.wsd
+++ b/docs/diagrams/src/motor_state.wsd
@@ -10,14 +10,12 @@ skinparam state {
 [*] --> Initialization
 
 state Initialization 
-    Initialization : Entry / Stop motor
-    Initialization : Do / Initialize internal parameters
-    Initialization : Do / Reset speed measurement
+    Initialization : Entry / Stop motor 
+    Initialization : Entry / Initialize internal parameters
+    Initialization : Entry / Reset speed measurement
     Initialization : Do / Wait 2 seconds
     Initialization : Exit / Send Boot-up message
 
-
-
 state Operational 
     Operational : Entry / LED on
     Operational : Do / Run speed controller
@@ -28,9 +26,7 @@ state Stopped
     Stopped : Do / Blink LED at 2 Hz
     Stopped : Exit / LED off
 
-
-
-Initialization --> Operational : initialization\n complete
+Initialization --> Operational : initialization complete
 Operational --> Initialization : reset
 Operational --> Stopped : fault detected
 
diff --git a/include/motor_controller.h b/include/motor_controller.h
index 770af32..d8cba89 100644
--- a/include/motor_controller.h
+++ b/include/motor_controller.h
@@ -42,13 +42,6 @@ public:
     );
 
 
-    // void set_states(
-    //     State* initialization,
-    //     State* operational,
-    //     State* stopped
-    // );
-
-
     void begin();
 
     void update();
diff --git a/src/main.cpp b/src/main.cpp
index 261b8e0..9a118e4 100644
--- a/src/main.cpp
+++ b/src/main.cpp
@@ -52,8 +52,7 @@ int main()
     motor.begin();
 
     sei();
-
-
+    
     // motor_controller.set_states(
     //     &initialization_state,
     //     &operational_state,
diff --git a/src/motor_controller.cpp b/src/motor_controller.cpp
index 3efb811..528f6b9 100644
--- a/src/motor_controller.cpp
+++ b/src/motor_controller.cpp
@@ -120,6 +120,20 @@ void MotorController::reset_speed_measurement()
     encoder.reset_speed_measurement();
 }
 
+<<<<<<< HEAD
+=======
+
+void MotorController::led_on()
+{
+    led.set_hi();
+}
+
+
+void MotorController::led_off()
+{
+    led.set_lo();
+}
+>>>>>>> 1d93605cc37a81ea974c12e62c082be19acdb715
 void MotorController::to_stopped()
 {
     transition_to(stopped_state);
@@ -129,6 +143,7 @@ bool MotorController::fault_active()
 {
     return motor.fault_active();
 }
+<<<<<<< HEAD
 
 void MotorController::led_on()
 {
@@ -140,3 +155,5 @@ void MotorController::led_off()
 {
    led.set_lo();
 }
+=======
+>>>>>>> 1d93605cc37a81ea974c12e62c082be19acdb715