Add tests to update