just added some testing code