Update main.py

Add call to main()