updated some testing code