Modified game restart conditions in  week2.py.