Beautify my code