cleaned up and added ML notebook