Add helper class and tests