can you edit the python code below according to the question in the screenshot.

import math

from matplotlib import pyplot as plt$,$ colors

from matplotlib.animation import FuncAnimation

import random

import numpy as np

from matplotlib.patches import Polygon

rand $=$ random.Random$()$

LANE$\_$WIDTH $=30$

NUM$\_$LANES $=2$

HIGHWAY$\_$LENGTH $=150$

HIGHWAY$\_$WIDTH $=40$

INTERARRIVAL $=6$

CAR$\_$SPEED $=3$

total$\_$cars $=0$

new$\_$cars $=0$

total$\_$in$\_$highway $=0$

total$\_$cars$\_$list $=[]$

new$\_$cars$\_$list $=[]$

colourmap $=$ colors.ListedColormap$(["$lightgrey$",$ "blue"$])$

normalizer $=$ colors.Normalize$($vmin$=0\mathrm{.}0,$ vmax$=1\mathrm{.}0)$

class Car:

def $\_\_$init$\_\_($self$,$ id$,$ highway$)$:

self.id $=$ id

self.active $=$ False

self.highway $=$ highway

self.lane $=$ None

self.position $=$ None

if rand.choice$([$True$,$ False$])$:

self.start$\_$lane $=0$

else:

self.start$\_$lane $=1$

self.start$\_$position $=0$

def enter$\_$highway$($self$,$ lane, position$)$:

if self.highway.enter$\_$highway$($self$,$ lane, position$)$:

self.active $=$ True

def step$($self$)$:

if self.position $$ HIGHWAY$\_$LENGTH $-1$:

self.highway.move$\_$car$($self$,$ self.lane, self.position $+1)$

else:

self.highway.leave$\_$highway$($self$)$

def $\_\_$str$\_\_($self$)$:

return $"$id: $\%$d lane: $\%$d position: $\%$d$"\%($self$.$id$,$ self.lane, self.position$)$

class Highway:

def $\_\_$init$\_\_($self$)$:

self.storage $=$ HighwayGrid$()$

self.bitmap $=[[0\mathrm{.}0$ for $\_$ in range$($HIGHWAY$\_$LENGTH$)]$ for $\_$ in range$($HIGHWAY$\_$WIDTH$)]$

self.time$\_$steps $=[0,0]$

def enter$\_$highway$($self$,$ car, lane, position$)$:

global total$\_$cars

if not lane $$ NUM$\_$LANES$)$:

print$("$Invalid lane!"$)$

return False

if self.storage.isoccupied$($lane$,$ position$)$:

print$("$Move rejected: occupied"$)$

return False

self.storage.add$\_$car$($lane$,$ position, car$)$

car.lane $=$ lane

car.position $=$ position

total$\_$cars $+=1$

return True

def leave$\_$highway$($self$,$ car$)$:

self.storage.remove$\_$car$($car$)$

def move$\_$car$($self$,$ car, lane, position$)$:

if position $$ HIGHWAY$\_$LENGTH:

if not self.storage.isoccupied$($lane$,$ position$)$:

self.storage.move$\_$car$($lane$,$ car.position, position, car$)$

car.position $=$ position

def run$\_$step$($self$,$ time$\_$step$)$:

for lane in range$($NUM$\_$LANES$)$:

if rand.random$()<mtext></mtext><mn>1</mn><mtext></mtext><mi>/</mi>$ INTERARRIVAL:

car $=$ Car$($total$\_$cars $+1,$ self$)$

car.enter$\_$highway$($lane$,$ car.start$\_$position$)$

self.time$\_$steps$[$lane$]=0$

total$\_$cars$\_$list.append$($total$\_$cars$)$

new$\_$cars$\_$list.append$($new$\_$cars$)$

self.time$\_$steps$[$lane$]+=1$

for car in self.storage.get$\_$list$()$:

if car.active:

if rand.random$()$ CAR$\_$SPEED:

car.step$()$

def refresh$\_$image$($self$)$:

# Clear the bitmap

self.bitmap $=[[0\mathrm{.}0$ for $\_$ in range$($HIGHWAY$\_$LENGTH$)]$ for $\_$ in range$($HIGHWAY$\_$WIDTH$)]$

for position in range$($HIGHWAY$\_$LENGTH$)$:

self.bitmap$[15][$position$]=1\mathrm{.}0$

self.bitmap$[25][$position$]=1\mathrm{.}0$

if position $\%5==0$:

self.bitmap$[20][$position$]=1\mathrm{.}0$

for car in self.storage.get$\_$list$()$:

lane $=$ car.lane

position $=$ car.position

self.bitmap$[17+$ lane $*5][$position$]=1\mathrm{.}0$

def isoccupied$($self$,$ lane, position$)$:

return self.storage.isoccupied$($lane$,$ position$)$

class HighwayGrid:

def $\_\_$init$\_\_($self$)$:

self.grid $=[[$None for $\_$ in range$($HIGHWAY$\_$LENGTH$)]$ for $\_$ in range$($NUM$\_$LANES$)]$

def isoccupied$($self$,$ lane, position$)$:

return self.grid$[$lane$][$position$]$ is not None

def add$\_$car$($self$,$ lane, position, car$)$:

if self.grid$[$lane$][$position$]$ is None:

self.grid$[$lane$][$position$]=$ car

return True

return False

def move$\_$car$($self$,$ lane, old$\_$position, new$\_$position, car$)$:

self.grid$[$lane$][$old$\_$position$]=$ None

self.grid$[$lane$][$new$\_$position$]=$ car

def remove$\_$car$($self$,$ car$)$:

lane $=$ car.lane

position $=$ car.position

self.grid$[$lane$][$position$]=$ None

def get$\_$list$($self$)$:

cars $=[]$

for lane in self.grid:

for car in lane:

if car is not None:

cars.append$($car$)$

return cars

hw $=$ Highway$()$

display $=$ plt$.$figure$($figsize$=(15,4))$

image $=$ plt$.$imshow$($hw$.$bitmap, cmap$=$colourmap, norm$=$normalizer, animated$=$True$)$

t $=0$

def updatefigure$(*$args$)$:

global t

t $+=1$

hw$.$run$\_$step$($t$)$

hw$.$refresh$\_$image$()$

image.set$\_$array$($hw$.$bitmap$)$

return image,

anim $=$ FuncAnimation$($display$,$ updatefigure, frames$=2000,$ interval$=50,$ blit$=$True, repeat$=$False$)$

plt$.$show$()$