Add method to set testing