still fixing ipynb output