Update main.py

Fix pawnlogic