Prettified code

Made layout of code more understandable. Minor change.