Added method to test