lets move the setup code into a module I can reuse between notebooks