updating unit testing