add update unit tests