added testing back in