add a helper class to test logic and output