Update pages/Fastest_Lap.py

pages/Fastest_Lap.py

@@ -1,6 +1,6 @@
 import streamlit as st
-from repo_directory import Fastest_Lap
-from repo_directory import button, utils
+from repo_directory.v2 import Fastest_Lap, utils
+from repo_directory import button
 
 
 YEAR_SELECTED = st.selectbox(
@@ -15,9 +15,4 @@ SESSION = st.selectbox(
     'Select Session',
     utils.get_sessions(YEAR_SELECTED, RACE_SELECTED))
 
-
-df, f1session, total_time, driver = Fastest_Lap.get_fastest_laps(YEAR_SELECTED,RACE_SELECTED, SESSION )
-
-
-
-Fastest_Lap.plot(df, f1session,SESSION,  total_time, driver)
+_, df = Fastest_Lap.plot(YEAR_SELECTED, RACE_SELECTED, SESSION)