Clean up

Clean up of module file, removing unused functions.
Added two block example notebook to showcase use of the package.

To do: add documentation for functions and readme file.