Update screenshot of python code

Fix indentation error