Re-indented the Python code