Small bug fix and a notebook for playing around.