Create three mice problem.py