Fixed notebook script and added the raw python