Tetris Python Code Without Loops