wip tests
add tests