Adding some Python code