Move unit test to its own folder