fix bug in readme
add tests