Tetris example improvements