add some more tests
update tests