udpating testing framework