add tests
Update test