written whack a mole.py