add tests wip