with the new design, should we move the test code to semarate example/(unit)test files?