Tetris in Python