Yeah that's good.

Furthermore, using that is a well-ordered set, your proof doesn't need the axiom of choice.

You could make it shorter considering "Prove that a non-finite subset of N is countable" instead of "Prove every subset of N is either finite or countable", but that changes almost nothing.