Begin adding Python code