Add tests
Update