Set rotation based on where player is facing

Hey guys

So I have a 2D prototype I am testing. I have a working moving and turning ship. My issue however is:

How can I make my ship "continue" to face the direction I moved?

Currently when I move my ship turns into that direction however when I stop moving it resets looking straight ahead.

I don't want that. Any help would be appreciated

Code: Look at the Move() and Turning() functions

 public float speed;
     
     public GameObject shot;
     public Transform shotSpawn;
     public float fireRate;
 
     private Vector3 movement;
     private Rigidbody rb;
     private AudioSource audioFire;
     private float nextFire;
 
     void Awake ()
     {
         audioFire = GetComponent<AudioSource>();
         rb = GetComponent<Rigidbody>();
     }
 
     void Update()
     {
         if(Input.GetMouseButtonDown(0) && Time.time > nextFire)
         {
             nextFire = Time.time + fireRate;
             Instantiate(shot, shotSpawn.position, shotSpawn.rotation);
             audioFire.Play();
         }
     }
 
     void FixedUpdate ()
     {
         float h = Input.GetAxis("Horizontal");
         float v = Input.GetAxis("Vertical");
 
     
         Move(h, v);
         Turning(h,v);
     }
 
     void Move(float h, float v)
     {
         movement.Set(h, 0.0f, v);
 
         movement = movement.normalized * speed * Time.deltaTime;
 
         rb.MovePosition(transform.position + movement);
     }
 
     void Turning(float h, float v)
     {
         //this makes it rotate and turn towards the direction I am moving
         transform.forward = Vector3.Normalize(new Vector3(h, 0f, v));
     }
 }