Updated notebook to reflect changes in helpers