Updated license info
removed unnecesary imports
Updated notebook
clean some noise