Update main.py

Modify image definitions
Change pieces' pos to tuple