Modified the code a little bit

Indentation was added to the code