Python code for presentation