Add simple test framework
Minor fix in simple test framework