Add minimal testing code