Fix indentation and style

Немного причесал код, под свое видение. Интересно мнение.