slight update to main nb and significant first pass at testing notebook