Two changes to ipynb setup

This is an attempt to fix alexdebrie/spark-flask#20 and alexdebrie/spark-flask#15. I'll test it and (hopefully) close the issues as soon as this is merged.