Added wait time in main.py