Change the prototype to use the simpy library