Traffic Flow Prediction Python Snake