Agent Based Model In Python Code