Write simple test framework