Update main.py

Now uses Piece class