add more tests and readme