Add testing unit