WIP Add tests