A simple version of Tetris in Python