Beautify the code

Beautify the code