from simple import SIMPLE problem = SIMPLE((1, 2), (0.7, 0.5), 0.02, 120) problem.prep() for i in range(3000): print(f'Iteration {i}') problem.iterate() if i % 5 == 0 or i == 1: problem.plot(True, 2)