WIP add some tests