Update main.py

コードの簡素化