Código de la simulación en formato .py