Add tests for updates