Carry momentum from a move

I am making a move where the character dives, lands on his belly and can then do a roll jump forward to get back up. But when he lands on his belly I want him to slide a little with the momentum from the dive.

Here is a video of the move

Currently, as he lands, all momentum stops. I don't know how to fix it. There is no particular reason that I use addForce for one of them and Transform.Position for the other.

 void update () {
 //dive 
 if (diveHappened)    
         {
             transform.position += new Vector3(1, 0, 0) * Time.deltaTime * divespeed;  
         }
 
 
         if (LieOnGround && Input.GetButtonDown("Jump"))
         {
 
             rb.AddForce(new Vector2(2, 2) * power, ForceMode2D.Impulse);
         } 
 }