added everything for the notebook hopefully