Beautify code

Just beautifying