Simple tetris in python