Minor refactoring
Corrected indentation
Fixing indentation