diff --git a/verkehrssimulation/verkehr_sk.py b/verkehrssimulation/verkehr_sk.py
index 755bb2f..17e2085 100644
--- a/verkehrssimulation/verkehr_sk.py
+++ b/verkehrssimulation/verkehr_sk.py
@@ -2,6 +2,7 @@
 
 import numpy as np
 import matplotlib.pyplot as plt
+import time
 
 GRIDSIZE = 300
 INIT_CELLS = int(GRIDSIZE*0.1)
@@ -16,6 +17,12 @@ def get_distance(grid, i):
     """
     Insert your code here
     """
+
+    if i == GRIDSIZE-1:
+        dist = (grid[i], grid[0])
+    else:
+        dist = (grid[i], grid[i+1])
+
     return dist
 
 # state transition t -> t + dt
@@ -23,6 +30,12 @@ def update(grid_old, grid_new):
     for i in range(GRIDSIZE):
         dist = get_distance(grid_old, i)
         
+        if not -1 in dist:
+            dist = dist[1] - dist[0]
+
+        print(dist)
+        time.sleep(0.5)
+
         """ Update cars according rules using value 'dist' """
         """
         Insert your code here