python code for tony stark