Agent Based Models Python