Update main.py

Define black pieces