Solow Growth Model, tutorial 3, Python version