Program writing with Python