edited tetris.py so that it is easier to edit