Added some testing utilities. Trying to figure out ML bug