Update tetris.py