added some testing code