diff --git a/src/index.ts b/src/index.ts
index 00ab974..0edb967 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -41,11 +41,11 @@ async function visitCityPages(): Promise<void> {
 
             console.log("Logo click")
             if (!await UIActions.gotoHome(driver)) throw `${city} failed`;
-            await WebDriverUtils.wait();
+            await WebDriverUtils.wait(driver);
 
             console.log("Exec Search")
             if (!await UIActions.execSearch(driver, city)) throw `${city} failed`;
-            await WebDriverUtils.wait(5);
+            await WebDriverUtils.wait(driver);
 
             console.log("Click See all")
             let seeAllUrl = await UIActions.getSeeAllUrl(driver);
@@ -85,7 +85,7 @@ async function visitCityPages(): Promise<void> {
             // click museum
             console.log("Click Museum link");
             if (!await UIActions.clickMuseumsLink(driver)) throw `${city} failed`;
-            await WebDriverUtils.wait();
+            await WebDriverUtils.wait(driver);
 
             let page = 1;
             while (1) {
@@ -93,12 +93,12 @@ async function visitCityPages(): Promise<void> {
                 // get list of museums
                 console.log("Get list of museums");
                 const museumElms = await UIActions.getMusiums(driver);
-                await WebDriverUtils.wait();
+                await WebDriverUtils.wait(driver);
 
                 for (const listItem of museumElms) {
 
                     await listItem.click();
-                    await WebDriverUtils.wait();
+                    await WebDriverUtils.wait(driver);
 
                     windows = await driver.getAllWindowHandles();
                     for (const handle of windows) {
@@ -115,10 +115,10 @@ async function visitCityPages(): Promise<void> {
 
                     museumWindow && await driver.switchTo().window(museumWindow);
                     await driver.close();
-                    await WebDriverUtils.wait();
+                    await WebDriverUtils.wait(driver);
 
                     attactionsWindow && await driver.switchTo().window(attactionsWindow);
-                    await WebDriverUtils.wait();
+                    await WebDriverUtils.wait(driver);
 
                 }
 
@@ -127,7 +127,7 @@ async function visitCityPages(): Promise<void> {
                 if (page > 10) break;
 
                 UIActions.clickPagination(driver, page);
-                await WebDriverUtils.wait();
+                await WebDriverUtils.wait(driver);
 
             }
 
diff --git a/src/lib/UIActions.ts b/src/lib/UIActions.ts
index b2b8ac6..ca2c21b 100644
--- a/src/lib/UIActions.ts
+++ b/src/lib/UIActions.ts
@@ -21,9 +21,9 @@ export async function execSearch(driver: WebDriver, city: string): Promise<boole
         await searchInput.sendKeys(city);
 
         // Submit the search (press Enter)
-        await WebDriverUtils.wait(2);
+        await WebDriverUtils.wait(driver);
         await searchInput.sendKeys('\uE007'); // Unicode for Enter key
-        await WebDriverUtils.wait(5); // Wait 5 seconds before next city
+        await WebDriverUtils.wait(driver); // Wait 5 seconds before next city
 
         return true;
     } catch (e) {
diff --git a/src/lib/utils.ts b/src/lib/utils.ts
index 289aba8..f49089a 100644
--- a/src/lib/utils.ts
+++ b/src/lib/utils.ts
@@ -14,10 +14,36 @@ export class WebDriverUtils {
      * @param seconds Number of seconds to wait
      * @returns Promise that resolves after the specified time
      */
-    static async wait(seconds: number = 3): Promise<void> {
-        seconds = Math.floor(Math.random() * 1000) % 3 + 3;
-        console.log(`Waiting for ${seconds} seconds...`);
-        return new Promise(resolve => setTimeout(resolve, seconds * 1000));
+    static async wait(driver?: WebDriver): Promise<void> {
+        const seconds = Math.floor(Math.random() * 1000) % 3 + 3;
+        console.log(`Scrolling to bottom for ${seconds} seconds...`);
+    
+        const endTime = Date.now() + seconds * 1000;
+    
+        let scrollCounter = 0;
+
+        while (Date.now() < endTime) {
+            try {
+                if(driver){
+                    if(scrollCounter < 4){
+                        await driver.executeScript(`
+                            window.scrollBy(0, window.innerHeight);
+                        `);
+                    }else{
+                        await driver.executeScript(`
+                            window.scrollTo(0, 0);
+                        `); 
+                    }
+
+                    scrollCounter++;
+                }
+            } catch (error) {
+                console.warn('Scroll failed:', error);
+            }
+    
+            // Wait a little between scrolls
+            await new Promise(resolve => setTimeout(resolve, 500));
+        }
     }
 
     /**