Feels like Goku is being a bit selfish by staying on a dying planet just to finish his fight with Frieza, but I also get that he needs to end this threat once and for all and seeing him rise out of the magma and send Gohan away was so intense and it really shows his growth as a warrior