Add helper code