improvement

add some style and comments