Improve code.
Indentation fix.