Some changes I needed to run this notebook