Add notebook + fixes