Add testing setup