Improving code formatting
Fixing indentation