Correct indentation in Python code