Adding update unit tests