iron man python code