Update tetris.py

more from porting over from 2.7 to 3.x