cleanedup notebook slightly