Added python script and notebook