made changes outlined in comments of pervious notebook commit