Merge pull request #102 from djchou/patch-1

Update for jupyter compatibility