add some Python code