added new testing funcitonality