Beautify some code