Python Tetris Part 11 (Refactoring)