Update main.py

Test with long time connection