A model of a simple economy using Python