Converted Notebook to py File