fix #2
reformat code