Trying to write tetris clone on Python