added tests
update tests