updated tests
updated tests