Small modification of python indentation