Fixed minor bug in #163; modified notebook also