fix notebook a bit