I think there are both pros and cons to him staying but him leaving was the main catalyst for his character growth/arc

I think Alexa actually said she doesn’t know if he could have the same amount of character growth if he had stayed in Stars Hollow Alex