Implement tests (WIP)