updating unit test WIP