lightly refactor code
add readme