Add helper methods for testing suite