Update main.py

new content added in line no 308