Add method to test