diff --git a/dissipating-waves/README.md b/dissipating-waves/README.md
index aaba351..3cd9996 100644
--- a/dissipating-waves/README.md
+++ b/dissipating-waves/README.md
@@ -102,7 +102,7 @@ If you would like to cite this work, please use the following citation:
 @article{wu2017flow,
   title={Dissipating stop-and-go waves in closed and open networks via deep reinforcement learning},
   author={Kreidieh, Aboudy and Wu, Cathy and Bayen, Alexandre M},
-  journal={ICCPS},
+  journal={ITSC},
   year={2018}
 }
 ```
diff --git a/flow-framework/utils.py b/flow-framework/utils.py
index b009a10..82947d7 100644
--- a/flow-framework/utils.py
+++ b/flow-framework/utils.py
@@ -25,6 +25,9 @@ def additional_command(self):
                     in [FollowerStopper, PISaturation]:
                 # set the color as red
                 self.k.vehicle.set_color(veh_id, color=(255, 0, 0))
+                # set the leader as cyan
+                leader = self.k.vehicle.get_leader(veh_id)
+                self.k.vehicle.set_color(leader, color=(0, 255, 255))
                 # if below automated time, use IDM to control vehicle dynamics
                 if self.time_counter < self.automated_time:
                     accel = self.controllers[veh_id].get_accel(self)