beautify code! =)