minor fix: updated readme
tested notebook a bit