Tests and minor correction in TreeCPD