fixed readme
fixed the comment
refactor #4