Python remake of Tetris using PyGame